#include <dbg.h>

#include <cassert>
using namespace std;

class Solution {
public:
    long long maximumSubarraySum(vector<int>& nums, int k) {
        long long sum = 0;
        long long res = 0;
        unordered_map<int, int> cnt;
        for (int i = 0; i < nums.size(); ++i) {
            sum += nums[i];  // 更新
            ++cnt[nums[i]];
            if (i < k - 1) {  // 不够k个
                continue;
            }
            if (cnt.size() == k) {  // 满足条件
                res = max(sum, res);
            }
            sum -= nums[i - k + 1];  // 移动窗口
            if (--cnt[nums[i - k + 1]] == 0) {
                cnt.erase(nums[i - k + 1]);
            }
        }
        return res;
    }
};

int main() {
    Solution s;

    vector<int> nums1 = {1, 5, 4, 2, 9, 9, 9};
    vector<int> nums2 = {4, 4, 4};

    int k1 = 3;
    int k2 = 3;

    long long res1 = 15;
    long long res2 = 0;

    assert(s.maximumSubarraySum(nums1, k1) == res1);
    assert(s.maximumSubarraySum(nums2, k2) == res2);

    return 0;
}